# encoding=utf-8
'''
目标要求有四个功能：1.登录和退出 2.查询余额 3.取钱 4.存钱
分为三个步骤：
（1）存放账号信息数据
（2）实现四个功能
（3）完成交互

'''
#导入按键函数
import keyboard
#账号信息数据源
d1={'账号':'12345',"密码":'12345',"金额":'100'}
d2={'账号':'123456','密码':'123456',"金额":'1000'}
d3={'账号':'1234567','密码':'1234567',"金额":'10000'}
account_list=[d1,d2,d3]
#登录界面
print('='*20,'欢迎登陆','='*20)
#定义登录函数
def login(number,password):
    for i in account_list:
        if number==i["账号"] and password==i["密码"]:
            return True
            break
        else:
            pass
#定义查询函数
def find():
    for i in account_list:
        money=int(i['金额'])
        if number==i["账号"] and password==i["密码"] and money==int(i['金额']):
            print("当前余额为{}元".format(money))
            return money
#定义取钱函数
def getmoney():
    for i in account_list:
        money=int(i['金额'])
        if number==i["账号"] and password==i["密码"] and money==int(i['金额']):
            print("当前余额为{}元".format(money))
            x=int(input("请输入取款金额："))
            if money>=x:
                money=money-x
                print("已取款{}元，当前余额为{}元".format(x,money))
                continue
            else:
                print("余额不足！")
                break

#定义存钱函数
def savemoney():
    for i in account_list:
        money=int(i['金额'])
        if number==i["账号"] and password==i["密码"] and money==int(i['金额']):
            print("当前余额为{}元".format(money))
            y=int(input("请输入存款金额："))
            money=money+y
            print("已存金额{}元,余额{}元".format(y,money))

#设置交互界面
number=input('请输入账号:')
password=input('请输入密码:')
money=account_list[2]
#设置循环判断所输入账号密码是否正确
while True:
    if login(number,password)==True:
        print('='*20,'登陆成功','='*20)
        print('''
    1.查询余额
    2.取钱
    3.存钱''')
        print('')
        break
    else:
        print('=' * 20, '账号或密码错误', '=' * 20)
        print('请重新输入')
        number = input('请输入账号:')
        password = input('请输入密码:')
while True:
    j=input('请选择您所需要的服务/按6退出：')
    if int(j) == 1:
        find()
    if int(j) == 2:
        getmoney()
    if int(j) == 3:
        savemoney()
    if int(j) == 6 :
        quit()
    if int(j)!=1 and int(j)!=2 and int(j)!=3 and int(j)!=6:
        print("输入错误请重新选择！")




